/*
 * @lc app=leetcode.cn id=188 lang=cpp
 * @lcpr version=30204
 *
 * [188] 买卖股票的最佳时机 IV
 */


// @lcpr-template-start
using namespace std;
#include <algorithm>
#include <array>
#include <bitset>
#include <climits>
#include <deque>
#include <functional>
#include <iostream>
#include <list>
#include <queue>
#include <stack>
#include <tuple>
#include <unordered_map>
#include <unordered_set>
#include <utility>
#include <vector>
// @lcpr-template-end
// @lc code=start
class Solution {
public:
    int maxProfit(int k, vector<int>& prices) {
        int n = prices.size();
        vector memo(n, vector<array<int, 2>>(k + 1, {-1, -1})); // -1 表示还没有计算过
        auto dfs = [&](this auto&& dfs, int i, int j, bool hold) -> int {
            if (j < 0) {
                return INT_MIN / 2; // 除 2 防止溢出
            }
            if (i < 0) {
                return hold ? INT_MIN / 2 : 0;
            }
            int& res = memo[i][j][hold]; // 注意这里是引用
            if (res != -1) { // 之前计算过
                return res;
            }
            if (hold) {
                return res = max(dfs(i - 1, j, true), dfs(i - 1, j - 1, false) - prices[i]);
            }
            return res = max(dfs(i - 1, j, false), dfs(i - 1, j, true) + prices[i]);
        };
        return dfs(n - 1, k, false);
    }
};
// @lc code=end



/*
// @lcpr case=start
// 2\n[2,4,1]\n
// @lcpr case=end

// @lcpr case=start
// 2\n[3,2,6,5,0,3]\n
// @lcpr case=end

 */

